package com.cb2.algorithm.leetcode;

/**
 * <a href="https://leetcode.cn/problems/insert-into-a-binary-search-tree/">二叉搜索树中的插入操作(Insert into a Binary Search Tree)</a>
 * <p>给定二叉搜索树（BST）的根节点 root 和要插入树中的值 value ，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据保证新值和原始二叉搜索树中的任意节点值都不同。</p>
 * <p>注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果 。</p>
 * <p>
 * <b>示例：</b>
 * <pre>
 * 示例 1：
 *      输入：root = [4,2,7,1,3], val = 5
 *                  4
 *                 / \
 *                2   7
 *               / \
 *              1   3
 *      输出：[4,2,7,1,3,5]
 *                   4                5
 *                 /   \            /   \
 *                2     7   或者    2     7
 *               / \   /          / \
 *              1   3 5          1   3
 *                                    \
 *                                     4
 *
 * 示例 2：
 *      输入：root = [40,20,60,10,30,50,70], val = 25
 *      输出：[40,20,60,10,30,50,70,null,null,25]
 *
 * 示例 3：
 *      输入：root = [4,2,7,1,3,null,null,null,null,null,null], val = 5
 *      输出：[4,2,7,1,3,5]
 * </pre>
 * </p>
 * <p>
 * <b>提示：</b>
 * <ul>
 *     <li>树中的节点数将在 [0, 10^4]的范围内。</li>
 *     <li>-10^8 <= Node.val <= 10^8</li>
 *     <li>所有值 Node.val 是 独一无二 的。</li>
 *     <li>-10^8 <= val <= 10^8</li>
 *     <li>保证 val 在原始BST中不存在。</li>
 * </ul>
 * </p>
 *
 * @author c2b
 * @since 2025/2/11 13:55
 */
public class LC0701InsertIntoBinarySearchTree_M {
    static class Solution {
        public TreeNode insertIntoBST(TreeNode root, int val) {
            //return bfs(root, val);
            return dfs(root, val);
        }

        private TreeNode bfs(TreeNode root, int val) {
            if (root == null) {
                return new TreeNode(val);
            }
            TreeNode currNode = root;
            while (true) {
                if (val < currNode.val) {
                    if (currNode.left == null) {
                        currNode.left = new TreeNode(val);
                        break;
                    } else {
                        currNode = currNode.left;
                    }
                } else if (val > currNode.val) {
                    if (currNode.right == null) {
                        currNode.right = new TreeNode(val);
                        break;
                    } else {
                        currNode = currNode.right;
                    }
                }
            }
            return root;
        }

        public TreeNode dfs(TreeNode currNode, int val) {
            // 如果当前节点为根的树为空，说明找到了插入位置，创建新节点并返回
            if (currNode == null) {
                return new TreeNode(val);
            }
            // 如果要插入的值小于当前节点的值，在左子树中进行插入操作
            if (val < currNode.val) {
                currNode.left = dfs(currNode.left, val);
            }
            // 如果要插入的值大于等于当前节点的值，在右子树中进行插入操作
            else if (val > currNode.val) {
                currNode.right = dfs(currNode.right, val);
            }
            // 返回当前节点，保持树的结构
            return currNode;
        }
    }

    public static void main(String[] args) {
        /*
                     40
                   /   \
                 20     60
                / \     / \
              10  30   50 70
         */
        TreeNode node = new TreeNode(40);
        node.left = new TreeNode(20);
        node.left.left = new TreeNode(10);
        node.left.right = new TreeNode(30);
        node.right = new TreeNode(60);
        node.right.left = new TreeNode(50);
        node.right.right = new TreeNode(70);
        Solution solution = new Solution();
        Printer.printTreeNode(solution.insertIntoBST(node, 25));
    }

}
